import unittest
from emr.emr_cluster_provision import EmrClusterProvision


class EmrClusterProvisionTestCase(unittest.TestCase):
    def test_something(self):
        p = EmrClusterProvision(
            master_type='c4.large',
            instance_type='c4.large',
            instance_count=2,
            driver_memory='5000m',
            executor_memory='5000m',
            bootstrap_script='launch_cluster_emr_bootstrap.sh')
        print(p)


if __name__ == '__main__':
    unittest.main()
